package com.imyours.message;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.imyours.model.message.MessageEntity;
import com.imyours.model.user.UserEntity;

/**页面缓存对象
 * @author SLevenc
 *
 */
public class MsgCache {
	
	Map<Long,List<MessageEntity>> msgMap = null;
	
	/**往缓存中添加信息
	 * @param msgEntity
	 */
	public void addMessage(MessageEntity msgEntity){
		if(msgEntity!=null){
			List<MessageEntity> msgs = msgMap.get(msgEntity.getReciver().getId());
			if(msgs == null){
				msgs = new ArrayList<MessageEntity>();
				msgMap.put(msgEntity.getReciver().getId(), msgs);
			}
			msgs.add(msgEntity);
		}
	}
	
	
	/**获取对应用户缓存的消息
	 * @param user
	 * @return
	 */
	public List<MessageEntity> getMessage(UserEntity user){
		List<MessageEntity> msgs = new ArrayList<MessageEntity>();
		if(user!=null){
			if(msgMap.containsKey(user.getId())){
				msgs = msgMap.get(user.getId());
				msgMap.remove(user.getId());
			}
		}
		return msgs;
	}
	
	
	
	public MsgCache() {
		msgMap = new HashMap<Long,List<MessageEntity>>();
	}

}
